torch
accelerate
diffusers
wandb
ipdb
torchvision
transformers
hpsv2x==1.2.0
inflect
pydantic
trl
absl-py
ml-collections
peft
git+https://github.com/openai/CLIP.git